square pattern using while loop in python